import torch
a = torch.Tenser([
   [
       [1,2,3],
       [4,5,6]
   ],
    [
       [7,8,9],
       [10,11,12]
   ]
])
print(a)
print(a.shape)
b=a.view(3,-1)
print(b)